const cheerio = require('cheerio')

const addScript = (content) => {
    // console.log('解析出来的HTML', content)
    const $ = cheerio.load(content)
    /** 添加文件引用 */
    console.log("$('body #live2dResources')", $('body #live2dResources'))
    if(!$('body #live_init').length) {
        $('body').prepend(`
            <script id='live_init' src="./initLive2d.js"></script>
        `)
    }
    if(!$('body #live2d_2').length) {
        $('body').prepend(`
            <script id='live2d_2' src="./L2Dwidget.min.js"></script>
        `)
    }
    if (!$('body #live2d_1').length) {
        $('body').prepend(`
            <script id='live2d_1' src="./L2Dwidget.0.min.js"></script>
        `)
    }
    // console.log('添加之后的主文件', $.html())
    return $.html()
}

const deleteScript = (content) => {
    const $ = cheerio.load(content)
    if($('body #live_init').length) {
        $('body #live_init').remove()
    }
    if($('body #live2d_2').length) {
        $('body #live2d_2').remove()
    }
    if ($('body #live2d_1').length) {
        $('body #live2d_1').remove()
    }
    return $.html()
}

module.exports = {
    addScript,
    deleteScript
}